/*
 *    Copyright (c) 2018-2025, shenghua All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 * Neither the name of the pig4cloud.com developer nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 * Author: shenghua
 */

package com.chen.base.module.redis.config;

import cn.hutool.core.util.StrUtil;
import com.chen.base.utils.PropertyUtil;
import com.chen.base.utils.TenantContextHolder;
import lombok.extern.slf4j.Slf4j;
import org.springframework.cache.Cache;
import org.springframework.data.redis.cache.RedisCache;
import org.springframework.data.redis.cache.RedisCacheConfiguration;
import org.springframework.data.redis.cache.RedisCacheManager;
import org.springframework.data.redis.cache.RedisCacheWriter;
import org.springframework.lang.Nullable;

import java.time.Duration;
import java.util.Map;

/**
 * redis cache 扩展cache name自动化配置
 *
 * @author L.cm
 * @author ChenJinLong
 * <p>
 * cachename = xx#ttl
 */
@Slf4j
public class RedisCacheAutoManager extends RedisCacheManager {

	/** cacheNames的分隔符，用于分隔出缓存的 */
	private static final String SEPARATE = "#";
	/** 表示缓存永不过期 */
	private static final String ALWAYS = "-1";
	private static final int LENGTH = 2;

	public RedisCacheAutoManager(RedisCacheWriter cacheWriter,
								 RedisCacheConfiguration defaultCacheConfiguration,
								 Map<String, RedisCacheConfiguration> initialCacheConfigurations,
								 boolean allowInFlightCacheCreation) {
		super(cacheWriter, defaultCacheConfiguration, initialCacheConfigurations, allowInFlightCacheCreation);
	}

	@Override
	protected RedisCache createRedisCache(String name, @Nullable RedisCacheConfiguration cacheConfig) {
		if (StrUtil.isBlank(name) || !name.contains(SEPARATE) || name.split(SEPARATE).length < LENGTH) {
			long expireTime = PropertyUtil.getProperty("cache-expire-time", Long.class);
			cacheConfig = setTtl(cacheConfig, expireTime);
			return super.createRedisCache(name, cacheConfig);
		}

		String[] cacheArray = name.split(SEPARATE);
		if (!ALWAYS.equals(cacheArray[1])) {
			cacheConfig = setTtl(cacheConfig, Long.parseLong(cacheArray[1]));
		}
		return super.createRedisCache(cacheArray[0], cacheConfig);
	}

	/**
	 * 从上下文中获取租户ID，重写@Cacheable value 值
	 */
	@Override
	public Cache getCache(String name) {
		String tenantId = TenantContextHolder.getTenantId();
		tenantId = StrUtil.isNotBlank(tenantId) ? tenantId + StrUtil.COLON : "";

		return super.getCache(tenantId + name);
	}

	private RedisCacheConfiguration setTtl(RedisCacheConfiguration cacheConfig, long ttl){
		return cacheConfig != null ? cacheConfig.entryTtl(Duration.ofSeconds(ttl)) : null;
	}
}
